a18997b28f1ca6d7c38cd18fe7b063d37ccbe83f,georocket-server/src/test/java/io/georocket/service/ServiceTest.java,ServiceTest,testDiscover,#TestContext#,57

Before Change



  @Test
  public void testDiscover(TestContext context) {
    Vertx vertx = rule.vertx();
    Async async = context.async();

    Service.publishOnce(vertx, "A", "a").doOnError(context::fail).subscribe(v -> {
      Service.publishOnce(vertx, "A", "b").doOnError(context::fail).subscribe(v1 -> {
        Service.discover(vertx, "A").doOnError(context::fail).count().subscribe(count -> {
          context.assertEquals(2, count);
          async.complete();
        });
      });
    });
  }

  @Test

After Change


   */
  @Test
  public void testDiscover(TestContext context) {
    Vertx vertx = new Vertx(rule.vertx());
    Async async = context.async();

    ServiceDiscovery discovery = ServiceDiscovery.create(vertx);
    Service.publishOnce("A", "a", discovery, vertx)
      .flatMap(v -> Service.publishOnce("A", "b", discovery, vertx))
      .flatMap(v -> Service.discover("A", discovery, vertx))
      .count()
      .doOnTerminate(discovery::close)
      .subscribe(count -> {
        context.assertEquals(2, count);
        async.complete();
      }, context::fail);
  }

  /**